add_executable(KVCacheSegmentTest test_segment.cpp)

target_include_directories(KVCacheSegmentTest PRIVATE
    ${COMMON_DIR}/include
)

target_link_libraries(KVCacheSegmentTest PRIVATE
    GTest::gtest
    Threads::Threads
    common_lib
)

if(DEVICE STREQUAL "GPU")
    set(LINK_LIBS CUDA::cudart)
endif()

enable_testing()
add_test(NAME KVCacheSegmentTest COMMAND KVCacheSegmentTest)
